package com.liugui.springbootthreadpool.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

import java.util.concurrent.Executor;
import java.util.concurrent.ThreadPoolExecutor;

/**
 * @Author LiuGui
 * @Date 2022年01月15日 11:27:00
 **/
@Slf4j
@Configuration
@EnableAsync
public class ExecutorConfig {

    @Bean("asyncServiceExecutor")
    public Executor asyncServiceExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        // 获取jvm cpu 核心线程数
        int core = Runtime.getRuntime().availableProcessors();
        // 配置核心线程数
        executor.setCorePoolSize(core);
        // 配置最大线程数
        executor.setMaxPoolSize(core*2 + 1);
        // 配置队列大小
        executor.setQueueCapacity(40);
        // 配置线程池中的线程的名称前缀
        executor.setThreadNamePrefix("liugui-async-service-");
        // 设置拒绝策略
        // ThreadPoolExecutor.CallerRunsPolicy：由调用线程处理该任务
        // ThreadPoolExecutor.AbortPolicy:丢弃任务并抛出RejectedExecutionException异常。 默认策略
        // ThreadPoolExecutor.DiscardPolicy：也是丢弃任务，但是不抛出异常。
        // ThreadPoolExecutor.DiscardOldestPolicy：丢弃队列最前面的任务，然后重新尝试执行任务（重复此过程）
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        // 执行初始化
        executor.initialize();
        return executor;
    }

}
